import torch
import torch.nn as nn
from .PConv_model import Pconv_encoder,Pconv_decoder

class total_model(nn.Module):
    def __init__(self):
        super(total_model, self).__init__()
        self.encoder = Pconv_encoder()
        self.decoder = Pconv_decoder()

    def forward(self, input, mask):
        feature, mask = self.encoder(input, mask)
        feature, mask = self.decoder(feature, mask)
        return feature

